package 牛客算法篇;

import java.util.ArrayList;
import java.util.PriorityQueue;

/**
 * Created with IntelliJ IEDA.
 * Description:
 * User:86186
 * Date:2024-03-09
 * Time:8:25
 */

/**
 *
 牛客算法篇:BM46 最小的K个数
 描述
 给定一个长度为 n 的可能有重复值的数组，找出其中不去重的最小的 k 个数。例如数组元素是4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4(任意顺序皆可)。
 数据范围：0≤k,n≤100000≤k,n≤10000，数组中每个数的大小0≤val≤10000≤val≤1000
 要求：空间复杂度 O(n)O(n) ，时间复杂度 O(nlogk)O(nlogk)
 示例1
 输入：
 [4,5,1,6,2,7,3,8],4
 返回值：
 [1,2,3,4]
 说明：
 返回最小的4个数即可，返回[1,3,2,4]也可以
 */
public class GetLeastNumbers_Solution {
    public static void main(String[] args) {
        System.out.println(GetLeastNumbers_Solution(new int[]{4, 5, 1, 6, 2, 7, 3, 8}, 4));
    }
    public static ArrayList<Integer> GetLeastNumbers_Solution(int[] input, int k) {
        // write code here
        ArrayList<Integer> res = new ArrayList<Integer>();
        //排除特殊情况
        if(k == 0 || input.length == 0)
            return res;
        //大根堆
        PriorityQueue<Integer> q = new PriorityQueue<>((o1, o2)->o2.compareTo(o1));
        //构建一个k个大小的堆
        for(int i = 0; i < k; i++)
            q.offer(input[i]);
        for(int i = k; i < input.length; i++){
            //较小元素入堆
            if(q.peek() > input[i]){
                q.poll();
                q.offer(input[i]);
            }
        }
        //堆中元素取出入数组
        for(int i = 0; i < k; i++)
            res.add(q.poll());
        return res;
    }
}
